#!/usr/bin/env python3
"""
安装 ModelScope 的脚本
"""

import subprocess
import sys

def install_modelscope():
    """安装 ModelScope"""
    print("开始安装 ModelScope...")
    
    try:
        # 安装 modelscope
        subprocess.check_call([sys.executable, "-m", "pip", "install", "modelscope"])
        print("✅ ModelScope 安装成功")
        
        # 验证安装
        import modelscope
        print(f"✅ ModelScope 版本: {modelscope.__version__}")
        
    except subprocess.CalledProcessError as e:
        print(f"❌ ModelScope 安装失败: {e}")
        return False
    except ImportError:
        print("❌ ModelScope 安装后无法导入")
        return False
    
    return True

if __name__ == "__main__":
    if install_modelscope():
        print("🎉 安装完成！现在可以使用 ModelScope 下载模型了")
        print("示例命令: modelscope download --model unsloth/Qwen3-32B-unsloth-bnb-4bit --local_dir ./model")
    else:
        print("❌ 安装失败，请检查错误信息")
        sys.exit(1)
